python square root program